import os
import random

# 输入文件目录
input_dir = r'C:\Users\ASUS\Desktop\2024年新生赛出题\2024年新生赛出题(非大一组)\F题\F题_input'

# 确保目录存在
os.makedirs(input_dir, exist_ok=True)

# 小范围 N 值生成范围
small_n_values = [random.randint(1, 1000) for _ in range(4)]
file_indices = range(12, 16)

# 生成并保存小范围测试数据
for idx, n_value in zip(file_indices, small_n_values):
    with open(os.path.join(input_dir, f"{idx}.in"), "w") as f:
        f.write(f"{n_value}\n")

print("小范围测试数据已生成并保存在指定目录中。")
